import torchvision
from torch.utils.tensorboard import SummaryWriter

dataset_transforms = torchvision.transforms.Compose([
    torchvision.transforms.ToTensor()
])
train_set = torchvision.datasets.CIFAR10("./CIFAR10_dataset",train=True,transform=dataset_transforms,download=True)
test_set = torchvision.datasets.CIFAR10("CIFAR10_dataset",train=False,transform=dataset_transforms,download=True)

# print(test_set[0])
# img,target = test_set[0]
# print(img)
# print(test_set.classes)
# print(target)
# print(test_set.classes[target])
# img.show()



for i in range(10):
    img,target = test_set[i]
    writer = SummaryWriter("CIFAR10")
    writer.add_image("CIFAR0_tesr",img,i)
writer.close()